 python imitate_episodes.py --eval --task_name real_transfer_cube --ckpt_dir ckpt_diffusion --policy_class Diffusion  --chunk_size 32 --batch_size 32 --lr 1e-5 --seed 0 --num_steps 100 --resume_ckpt_path ckpt_diffusion/policy_best.ckpt
